package 中等.其他;

import java.util.Arrays;

/**
 * 给你两个整数 n 和 k ，请你构造一个答案列表 answer ，该列表应当包含
 * 从 1 到 n 的 n 个不同正整数，并同时满足下述条件：
 * 假设该列表是 answer = [a1, a2, a3, ... , an] ，那么列表 [|a1 - a2|
 * , |a2 - a3|, |a3 - a4|, ... , |an-1 - an|] 中应该有且仅有 k 个不同整数。
 * 返回列表 answer 。如果存在多种答案，只需返回其中 任意一种 。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/beautiful-arrangement-ii
 */
public class 优美的排列II_667 {

    public static void main(String[] args) {

        System.out.println(Arrays.toString(constructArray(5, 2)));

    }


    /**
     * 特殊情况分析
     * 例如 n = 5
     * 第一种情况，如果顺序排列
     * 1 2 3 4 5 ，那么 k = 1
     * 第二种情况，如果奇数位从最小值升序排列，偶数位从最大值降序排列
     * 1 5 2 4 3 ，那么 k = 4
     * 规律
     * 如果 k = 4 ，那么前 4 个位置是第二种情况排列，其余按第一种情况排列
     */
    public static int[] constructArray(int n, int k) {
        int[] ans = new int[n];
        int min = 1, max = n;
        int idx = 0;
        while (idx < k) {
            ans[idx] = min;
            min++;
            idx++;

            if (idx < k) {
                ans[idx] = max;
                max--;
                idx++;
            }
        }

        for (int i = 0; i < n - k; i++) {
            // 注意：后面的排列，必须全部差值为 1 ，如果前面一个位置是最大值，降序，最小值则升序
            if (ans[idx - 1] > max) {
                ans[idx] = max;
                max--;
            } else {
                ans[idx] = min;
                min++;
            }
            idx++;
        }
        return ans;
    }

}
